import torch
import torchvision.models as models
import torch.nn as nn
from util import Parameter
from EmbeddingModel import EmbeddingModel

def prepare50(prepared_path:str):

    model = EmbeddingModel()

    torch.save(model, prepared_path)
    # print(model)

if __name__=="__main__":
    prepare50("model/prepared/embedding_resnet50.pt")